#!/usr/bin/python3
import os
import shutil
import sys

def move_ko_files(source_dir, target_dir):
    # 确保目标目录存在
    if not os.path.exists(target_dir):
        os.makedirs(target_dir)

    # 遍历源目录及其所有子目录
    for root, dirs, files in os.walk(source_dir):
        for file in files:
            # 检查文件是否以.ko结尾
            if file.endswith('.out'):
                # 构造源文件的完整路径
                source_file = os.path.join(root, file)
                # 构造目标文件的完整路径（只包含文件名，不包含源目录结构）
                target_file = os.path.join(target_dir, file)
                # 移动文件
                shutil.move(source_file, target_file)
                print(f"Moved {source_file} to {target_file}")

if __name__ == "__main__":
    # 检查是否提供了目标目录作为参数
    if len(sys.argv) != 2:
        print("Usage: python script.py <target_directory>")
        sys.exit(1)

    target_dir = sys.argv[1]
    current_dir = os.getcwd()

    # 调用函数移动.ko文件
    move_ko_files(current_dir, target_dir)

    print("All .ko files have been moved to the specified directory.")
